
from monitor_tools.plot_memory import parse_args, plot_data
import os

if __name__ == "__main__":
    args = parse_args()

    for log_file in args.log_files:
        print("deal with log_file : ", log_file)
        log_name = os.path.basename(log_file)
        fig_name = log_name.split('.')[0] + '.png'
        plot_data(log_file, fig_name, xpu_type='GPU')
